#pragma once

#include <signal.h>
#include <unistd.h>
#include <cstdlib>
#include <string>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

#define Setsid_Err 1
#define Chdir_Err 2
const std::string cwd_path = "/";
const std::string dev_null_path = "/dev/null";

void Daemon(bool ischdir, bool isclose)
{
    // 1.屏蔽可能导致进程退出的信号
    signal(SIGPIPE, SIG_IGN);
    signal(SIGCHLD, SIG_IGN);

    // 2.不让自己成为组长
    if (fork() > 0)
        exit(0);

    // 3. 设置让自己成为一个新的会话， 后面的代码其实是子进程在走
    if (setsid() < 0)
        exit(Setsid_Err);

    // 4. 每一个进程都有自己的CWD，是否将当前进程的CWD更改成为 / 根目录
    if (ischdir)
    {
        if (chdir(cwd_path.c_str()) < 0)
            exit(Chdir_Err);
    }

    // 5. 已经变成守护进程啦，不需要和用户的输入输出，错误进行关联了
    if (isclose)
    {
        close(0);
        close(1);
        close(2);
    }
    else
    {
        int fd = open(dev_null_path.c_str(), O_RDWR);
        if(fd > 0)
        {
            dup2(fd, 0);
            dup2(fd, 1);
            dup2(fd, 2);
            close(fd);
        }
    }
}